from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import MyUser  # 确保导入你的自定义用户模型

class MyUserAdmin(UserAdmin):
    # 你可以在这里自定义显示的字段
    model = MyUser
    list_display = ('username', 'email', 'first_name', 'last_name', 'is_staff', 'is_active', 'mobile', 'qq', 'weChat')
    list_filter = ('is_staff', 'is_active')
    search_fields = ('username', 'email', 'first_name', 'last_name', 'mobile', 'qq', 'weChat')
    ordering = ('username',)

    # 自定义字段集
    fieldsets = UserAdmin.fieldsets + (
        (None, {
            'fields': ('mobile', 'qq', 'weChat'),  # 添加自定义字段
        }),
    )

    # 添加用户时的字段集
    add_fieldsets = UserAdmin.add_fieldsets + (
        (None, {
            'fields': ('mobile', 'qq', 'weChat'),  # 添加自定义字段
        }),
    )

# 注册自定义用户模型
admin.site.register(MyUser, MyUserAdmin)
